# encoding: utf-8
"""
@author:  shaoqijun
@date:  2020/7/26
"""
from yacs.config import CfgNode as CN
import os
__C = CN()
cfg = __C
__C.CUDA = True
__C.DEVICES = [0,1]


#------------------------------------------
# DATA PARAMS
#------------------------------------------
__C.DATA = CN()
__C.DATA.ROOT = r'F:\DataSet\MNIST\data\MNIST\mnist_train'                        #训练数据目录
__C.DATA.RATION = 0.8                                                  #百分之80为训练集，百分之20为验证集

#-------------------------------------------
# TRAIN PARAMS
#-------------------------------------------
__C.TRAIN = CN()
__C.TRAIN.MEAN = [0.698, 0.692, 0.673]
__C.TRAIN.STD = [0.261, 0.266, 0.270]
__C.TRAIN.BATCH_SIZE = 256
__C.TRAIN.NUM_WORKERS = 4
__C.TRAIN.EPOCHS = 150
__C.TRAIN.LOG_DIR = './logs'
__C.TRAIN.RESNET18_ROOT = r'D:\proj\button\new_button\model\resnet18-5c106cde.pth'
__C.TRAIN.SNAPSHOT_DIR = './snapshot'

#-------------------------------------------
# val PARAMS
#-------------------------------------------
__C.VAL = CN()
#__C.VAL.DIR = r'D:\dataset\data\button\val'
__C.VAL.MEAN = [0.697, 0.690, 0.673]
__C.VAL.STD = [0.262, 0.269, 0.273]